#include <iostream>
#include <string>
#include <vector>
#include <functional>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <signal.h>
#include <stdlib.h>
#include <atomic>
using namespace std;

typedef function<void()> CallbackFunc;
vector<CallbackFunc> callbacks;

uint64_t count = 0;

void showCount()
{
    // 为了避免cout影响count的累加速度，这里可以暂时禁用cout的同步
    ios_base::sync_with_stdio(false);
    cout << "final count : " << count << endl;
    // 恢复cout的同步
    ios_base::sync_with_stdio(true);
}

void showLog()
{
    cout << "【日志功能】：触发了一次定时器事件" << endl;
}

void logUser()
{
    int pipefd[2];
    if (pipe(pipefd) == -1)
    {
        cerr << "Pipe creation failed." << endl;
        return;
    }

    pid_t childPid = fork();

    if (childPid == 0)
    {
        // 在子进程中复制标准输出到管道
        dup2(pipefd[1], STDOUT_FILENO);
        close(pipefd[0]); // 关闭读端
        execl("/usr/bin/who", "who", nullptr);
        _exit(1); // 如果 execl 失败，直接退出子进程
    }
    else if (childPid > 0)
    {
        close(pipefd[1]); // 关闭写端
        char buffer[4096];
        ssize_t nread;
        while ((nread = read(pipefd[0], buffer, sizeof(buffer) - 1)) > 0)
        {
            buffer[nread] = '\0';
            cout << buffer;
        }
        close(pipefd[0]);
        wait(nullptr); // 父进程等待子进程执行完成
    }
    else
    {
        cerr << "Failed to fork a new process." << endl;
    }
}

// SIGALRM信号处理器
std::atomic<bool> shouldLogUser(false);

void catchSig(int signum) {
    for (auto &f : callbacks) {
        f();
    }
    shouldLogUser = true; // 设置标志，后续在主循环中处理
    alarm(1); // 重新设置一个1秒后的定时器
}

int main(int argc, char *argv[]) {
    // 注册SIGALRM信号处理器
    signal(SIGALRM, catchSig);
    alarm(1);

    callbacks.push_back(showCount);
    callbacks.push_back(showLog);

    while (true) {
        ++count; // 累加计数

        if (shouldLogUser) {
            logUser(); // 在主循环中调用
            shouldLogUser = false; // 重置标志
        }
    }
    return 0;
}